numpy
matplotlib
pandas
scikit-learn
torch==1.9.0